package com.ruoyi.config;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.ruoyi.platform.utils.spring.SpringUtils;
import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Configuration;

import javax.annotation.PostConstruct;
import java.io.IOException;

/**
 * RabbitMq的消息确认机制
 */
@Configuration
@Slf4j
public class MqAckConfig implements RabbitTemplate.ConfirmCallback, RabbitTemplate.ReturnCallback {

    @Autowired
    private RabbitTemplate rabbitTemplate;

    @PostConstruct
    public void init() {
        rabbitTemplate.setConfirmCallback(this);
        rabbitTemplate.setReturnCallback(this);
    }

    // 只确认消息是否正确到达Exchange中，无论正确与否，都会执行
    @Override
    public void confirm(CorrelationData correlationData, boolean ack, String cause) {
        if (ack) {
            log.debug("消息发送成功，消息内容：{}", correlationData);
        } else {
            log.error("消息发送失败，消息内容：{}", cause);
        }
    }

    // 消息没有正确到达队列时，将会被触发，如果正确到达队列，不会被执行。
    @Override
    public void returnedMessage(Message message, int replyCode, String replyText, String exchange, String routingKey) {
        ObjectMapper objectMapper = SpringUtils.getBean(ObjectMapper.class);
        try {
            log.debug("消息主体：{}", objectMapper.readValue(message.getBody(), Object.class));
        } catch (IOException e) {
            e.printStackTrace();
        }
        log.debug("描述：{}", replyText);
        log.debug("消息使用的路由键：{}", routingKey);
        log.debug("消息使用的交换器：{}", exchange);
    }

}
